package org.algorithm.Sort;

public class MaxHeap {
    public void heapify(int arr[], int n, int i) {
        int largest = i; // 初始化最大为根节点
        int left = 2 * i + 1; // 左子节点
        int right = 2 * i + 2; // 右子节点

        // 如果左子节点大于根节点
        if (left < n && arr[left] > arr[largest]) {
            largest = left;
        }

        // 如果右子节点大于当前最大节点
        if (right < n && arr[right] > arr[largest]) {
            largest = right;
        }

        // 如果最大不是根节点
        if (largest != i) {
            int swap = arr[i];
            arr[i] = arr[largest];
            arr[largest] = swap;

            // 递归地调整受影响的子树
            heapify(arr, n, largest);
        }
    }

    // 构建大顶堆
    public void buildMaxHeap(int arr[]) {
        int n = arr.length;

        // 从最后一个非叶子节点开始，向上构造大顶堆
        for (int i = n / 2 - 1; i >= 0; i--) {
            heapify(arr, n, i);
        }
    }

    // 打印堆
    public static void printArray(int arr[]) {
        int n = arr.length;
        for (int i = 0; i < n; ++i)
            System.out.print(arr[i] + " ");
        System.out.println();
    }

    // 主方法
    public static void main(String args[]) {
        int arr[] = {3, 1, 6, 5, 2};

        MaxHeap maxHeap = new MaxHeap();
        maxHeap.buildMaxHeap(arr);

        System.out.println("The Max Heap is ");
        printArray(arr);
    }
}
